import cv2

# 读取图像文件
lena = cv2.imread("./img/lena.png")

# 显示原始lena图像
cv2.imshow("lena", lena)

# 分离lena图像的蓝色通道
b = lena[:, :, 0]
# 分离lena图像的绿色通道
g = lena[:, :, 1]
# 分离lena图像的红色通道
r = lena[:, :, 2]

# 显示分离的蓝色通道图像
cv2.imshow("b", b)
# 显示分离的绿色通道图像
cv2.imshow("g", g)
# 显示分离的红色通道图像
cv2.imshow("r", r)

# 使用cv2.split函数分离lena图像的三个通道
b1, g1, r1 = cv2.split(lena)

# 显示使用cv2.split分离的蓝色通道图像
cv2.imshow("b1", b1)
# 显示使用cv2.split分离的绿色通道图像
cv2.imshow("g1", g1)
# 显示使用cv2.split分离的红色通道图像
cv2.imshow("r1", r1)

# 将lena图像的蓝色通道设置为0
lena[:, :, 0] = 0
# 显示修改后的lena图像，此时蓝色通道为0
cv2.imshow("lenab0", lena)

# 将lena图像的绿色通道设置为0
lena[:, :, 1] = 0
# 显示修改后的lena图像，此时蓝色和绿色通道为0
cv2.imshow("lenab0g0", lena)

# 将lena图像的红色通道设置为0
lena[:, :, 2] = 0
# 显示修改后的lena图像，此时所有通道都为0，图像应为全黑
cv2.imshow("lenabogoro", lena)

# 等待按键事件，任意键按下后关闭所有图像窗口
cv2.waitKey()
# 关闭所有cv2.imshow创建的窗口
cv2.destroyAllWindows()
